class Config:
    # visualization
    env = 'Googlenet'  # visdom env
    port = 8097
    plot_every = 40  # vis every N iter

    # data
    train_data_root = './data/train/'        # 训练集存放路径
    validate_data_root = './data/validation/'  # 验证集存放路径
    test_data_root = './data/test/'          # 测试集存放路径

    # 预训练模型
    load_model_path = None

    batch_size = 128   # batch size
    print_freq = 10  # print info every N batch

    result_file = 'result.csv'
    max_epoch = 100

    # param for optimizer
    weight_decay = 0.0005
    lr_decay = 0.95  # 1e-3 -> 1e-4
    lr = 1e-3

opt = Config()